import pandas as pd

# 读取Excel文件
df = pd.read_excel('20241104.xlsx', sheet_name='Sheet1')

# 读取部门ID映射的Excel文件
dept_id_df = pd.read_excel('企业及企业id.xlsx', sheet_name='Sheet1')

# 将读取的数据转换为字典
dept_id_mapping = dict(zip(dept_id_df['企业'], dept_id_df['企业ID']))

# 定义园区名称到ID的映射
park_id_mapping = {
    "湖南灵奥信息科技服务有限公司": 4,
    "湖北荆盛来网络科技有限公司": 5,
    "株洲市金财云智信息科技服务有限公司": 6,
    "湖南行维汽车科技服务有限公司": 7,
    "湖南金财聚惠信息科技服务有限公司": 8,
    "湖南金财卓可信息科技服务有限公司": 9,
    "湖南金财昌盛信息科技服务有限公司": 10,
    "湖南金财皆旺信息科技服务有限公司": 11,
    "湖南金财宽裕信息科技服务有限公司": 12,
    "湖南牧夫信息科技服务有限公司": 13,
    "湖南润垚信息科技服务有限公司": 14,
    "湖南联薪信息科技有限公司": 15,
    "湖南通友信息科技有限公司": 16,
    "湖南悦时信息科技有限公司": 17,
    "湖南灵奔信息科技服务有限公司": 18,
    "湖南金财合创信息科技服务有限责任公司": 19,
    "湖南联江信息科技有限公司": 20,
    "湖南鸿步信息科技有限公司": 21,
    "湖南焕强信息科技有限公司": 22,
    "湖南税猫科技有限公司": 23,
    "湖南灵达信息科技服务有限公司": 24,
    "湖南迅基信息科技有限公司": 25,
    "湖南齐锡信息科技有限公司": 26,
    "湖南茂新信息科技有限公司": 27,
    "湖南壹洽通网络科技有限公司": 28,
    "湖南丰时信息科技有限公司": 29,
    "永州云动信息技术有限公司": 30,
    "湖南奔奔炫网络科技有限公司": 31,
    "湖北省中农智谷信息科技有限公司": 32,
    "怀宁才汇网络科技有限公司": 33,
    "江西瑞满信息技术有限公司": 34
}

# 分割所属园区列，并展开为多行
df_expanded = df.assign(所属园区=df['所属园区'].str.split(',')).explode('所属园区').reset_index(drop=True)

# 添加新的列“园区ID”，并填充对应的ID
df_expanded['园区ID'] = df_expanded['所属园区'].map(park_id_mapping)

# 添加新的列“企业ID”，并填充对应的ID
df_expanded['企业ID'] = df_expanded['企业'].map(dept_id_mapping)

# 保存处理后的数据到新的Excel文件
df_expanded.to_excel('处理后的工作簿02.xlsx', index=False)

